package com.tanhua.sso.service;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.dubbo.config.annotation.Reference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.common.mapper.UserMapper;
import com.tanhua.common.pojo.User;
import com.tanhua.dubbo.server.api.HuanXinApi;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class UserService {

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private UserMapper userMapper;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    //一定要注意RedisTemplate<String,String>后面有String
    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    @Reference(version = "1.0.0")
    private HuanXinApi huanXinApi;

    public String login(String mobile, String code) {
        //定义查询的验证码的key，和之前存入的key格式相同
        String rediskey="CHECK_CODE_"+mobile;
        Boolean isNew=false;
        //校验验证码
        String redisData=redisTemplate.opsForValue().get(rediskey);
        if(!StringUtils.equals(redisData,code)){
            return null;//验证码错误
        }
        //删除验证码。以免重复登录时显示验证码尚未过期
        redisTemplate.delete(rediskey);
        QueryWrapper queryWrapper=new QueryWrapper();
        queryWrapper.eq("mobile",mobile);
        User user = userMapper.selectOne(queryWrapper);
        if(null==user){
            user=new User();
            user.setMobile(mobile);
            //md5加密初始密码
            user.setPassword(DigestUtils.md5Hex("123456"));
            //注册新用户
            userMapper.insert(user);
            //将标记置为true,表示新用户
            isNew=true;
            //注册环信用户
            Boolean result = huanXinApi.register(user.getId());
            if (!result) {
                //注册环信失败，记录日志
                log.error("注册环信用户失败~ userId = " + user.getId());
            }
        }

        //生成token相关数据
        Map<String,Object> claims=new HashMap<String,Object>();
        //只加id,手机号不能放
        claims.put("id",user.getId());

        //生成token
        String token= Jwts.builder()
                .setClaims(claims)//payload存放数据，不能放敏感数据，如密码等
                .signWith(SignatureAlgorithm.HS256,secret)//设置加密方法和盐值
                .setExpiration(new DateTime().plusHours(12).toDate())//设置过期时间，12小时后过期
                .compact();

        //这里注释掉try ..catch中给RocketMQ发消息
        /*try {
            //为以后的程序进行扩展（将成功的消息写入RocketMQ）
            //map集合注意数据类型
            Map<String,Object> msg=new HashMap<>();
            msg.put("id",user.getId());
            msg.put("date",System.currentTimeMillis());
            rocketMQTemplate.convertAndSend("tanhua-sso-login",msg);
        } catch (MessagingException e) {
            //避免rocketMQ崩掉
            log.error("RocketMQ发送消息失败",e);
        }*/
        return token+"|"+isNew;
    }

    public User queryUserByToken(String token){
        try {
            //通过token解析数据
            Map<String,Object> body=Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            User user=new User();
            //得到token里面的id
            user.setId(Long.valueOf(body.get("id").toString()));
        /*
           需要返回user对象中的mobile,需要查询数据库获取到mobile数据
           如果每次都查询数据库,必然会导致性能问题，需要对用户的手机号进行缓存操作
           数据缓存时，需要设置过期时间，过期时间要有token的时间一致
           如果用户修改了手机号,需要同步修改redis中的数据
        */
            String redisKey="TANHUA_USER_MOBILE_"+user.getId();
                if(this.redisTemplate.hasKey(redisKey)){//根据键获取手机号值
                    String mobile=this.redisTemplate.opsForValue().get(redisKey);
                    user.setMobile(mobile);
                }else{
                //查询数据库
                User user1 = userMapper.selectById(user.getId());
                user.setMobile(user1.getMobile());
                //将手机号写入redis中
                //在jwt中的过期时间的单位为秒
                Long timeout=Long.valueOf(body.get("exp").toString())*1000-System.currentTimeMillis();
                redisTemplate.opsForValue().set(redisKey,user1.getMobile(),timeout, TimeUnit.MILLISECONDS);
            }
            return user;
        } catch (ExpiredJwtException e) {
                log.error("token已经过期！token="+token,e);
        } catch (Exception e) {
            log.error("token不合法！token="+token,e);
        }
        return null;
    }

    /**
     * 更新手机号
     * @param userId
     * @param newPhone
     * @return
     */
    public Boolean updatePhone(Long userId, String newPhone) {
        //先查询新手机号是否已经注册，如果已经注册，就不能修改,查询mysql中的user表
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", newPhone);
        User user = this.userMapper.selectOne(queryWrapper);
        if(ObjectUtil.isNotEmpty(user)){
            //新手机号已经被注册
            return false;
        }

        user = new User();
        user.setId(userId);
        user.setMobile(newPhone);
        //修改user表的信息
        return this.userMapper.updateById(user) > 0;
    }

}
